988b475da808266f36f63898f43820c91bf2f471,dl4j-examples/src/main/java/org/deeplearning4j/examples/unsupervised/variational/VariationalAutoEncoderExample.java,VariationalAutoEncoderExample,main,#String[]#,36
Before Change
// List<float[]> temp = new ArrayList<>();
// temp.add(net.params().data().asFloat());
for( int i=0; i<nEpochs; i++ ){
net.fit(trainIter);
trainIter.reset();
// System.out.println(Arrays.toString(net.params().data().asFloat()));
// temp.add(net.params().data().asFloat());
After Change
for( int i=0; i<nEpochs; i++ ){
while(trainIter.hasNext()){
DataSet ds = trainIter.next();
net.fit(ds);
//Every N minibatches: collect the test set latent space values for later plotting
if(iterationCount++ % plottingLatentSpaceEveryNMinibatches == 0){